home *** CD-ROM | disk | FTP | other *** search
/ Komputer for Alle 2004 #2 / K-CD-2-2004.ISO / OpenOffice Sv / f_0397 / python-core-2.2.2 / lib / test / test_pickle.py < prev    next >
Encoding:
Python Source  |  2003-07-18  |  967 b   |  41 lines
  1. import pickle
  2. import test_support
  3. import unittest
  4. from cStringIO import StringIO
  5. from pickletester import AbstractPickleTests, AbstractPickleModuleTests
  6.  
  7. class PickleTests(AbstractPickleTests, AbstractPickleModuleTests):
  8.  
  9.     def setUp(self):
  10.         self.dumps = pickle.dumps
  11.         self.loads = pickle.loads
  12.  
  13.     module = pickle
  14.     error = KeyError
  15.  
  16. class PicklerTests(AbstractPickleTests):
  17.  
  18.     error = KeyError
  19.  
  20.     def dumps(self, arg, bin=0):
  21.         f = StringIO()
  22.         p = pickle.Pickler(f, bin)
  23.         p.dump(arg)
  24.         f.seek(0)
  25.         return f.read()
  26.  
  27.     def loads(self, buf):
  28.         f = StringIO(buf)
  29.         u = pickle.Unpickler(f)
  30.         return u.load()
  31.  
  32. def test_main():
  33.     loader = unittest.TestLoader()
  34.     suite = unittest.TestSuite()
  35.     suite.addTest(loader.loadTestsFromTestCase(PickleTests))
  36.     suite.addTest(loader.loadTestsFromTestCase(PicklerTests))
  37.     test_support.run_suite(suite)
  38.  
  39. if __name__ == "__main__":
  40.     test_main()
  41.